# 睡眠分期配置文件
# 定义标签常量
W = "W"
N1 = "N1"
N2 = "N2"
N3 = "N3"
REM = "REM"
UNKNOWN = "UNKNOWN"

# 定义数字到标签的映射字典
class_dict = {
    0: W,
    1: N1,
    2: N2,
    3: N3,
    4: REM,
    5: UNKNOWN
}


# 定义颜色映射（可选，用于可视化）
stage_colors = {
    W: "#FFCC00",  # 黄色
    N1: "#33CCFF",  # 浅蓝色
    N2: "#0066CC",  # 深蓝色
    N3: "#003366",  # 深蓝色
    REM: "#FF6699",  # 粉红色
    UNKNOWN: "#CCCCCC"  # 灰色
}
